package com.sali.DP;


/**
 * 小偷又发现了一个新的可行窃的地区。这个地区只有一个入口，我们称之为 root 。
 * <p>
 * 除了 root 之外，每栋房子有且只有一个“父“房子与之相连。一番侦察之后，聪明的小偷意识到“这个地方的所有房屋的排列类似于一棵二叉树”。 如果 两个直接相连的房子在同一天晚上被打劫 ，房屋将自动报警。
 * <p>
 * 给定二叉树的 root 。返回 在不触动警报的情况下 ，小偷能够盗取的最高金额 。
 */
public class LeetCode_337 {

    public int rob(TreeNode root) {
        int[] res = getRes(root);
        return Math.max(res[0], res[1]);
    }

    private int[] getRes(TreeNode root) {
        if (root == null) {
            return new int[]{0, 0};
        }

        int[] res1 = getRes(root.left);
        int[] res2 = getRes(root.right);
        int rob = root.val + res1[1] + res2[1];
        int not_rob = Math.max(res1[0], res1[1]) + Math.max(res2[0], res2[1]);
        return new int[]{rob, not_rob};
    }

}

class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;

    TreeNode() {
    }

    TreeNode(int val) {
        this.val = val;
    }

    TreeNode(int val, TreeNode left, TreeNode right) {
        this.val = val;
        this.left = left;
        this.right = right;
    }
}
